package algorithm;

import util.ListNode;

/**
 * Description:
 * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
 * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
 * 输出：7 -> 0 -> 8
 * 原因：342 + 465 = 807
 * @author: chengrongkai
 * Date: 2021/3/8
 */
public class Solution14 {

    public static void main(String[] args) {
        int[] nums1 = {9,9,9,9,9,9,9};
        int[] nums2 = {9,9,9,9};
        ListNode node1 = ListNode.getListNode(nums1);
        ListNode node2 = ListNode.getListNode(nums2);
        ListNode solution = solution(node1, node2);
        while (solution != null){
            System.out.println(solution.val);
            solution = solution.next;
        }
    }

    /**
     * 同样的思路，借助哨兵节点遍历，主要要注意节点值之和大于9的情况
     * @param node1
     * @param node2
     * @return
     */
    public static ListNode solution(ListNode node1,ListNode node2){
        ListNode node = new ListNode(-1);
        ListNode pre = node;
        boolean add = false;
        while (node1 != null && node2 != null){
            int val = node1.val + node2.val;
            if (add){
                val = val+1;
                add = false;
            }
            if (val >= 10){
                add = true;
                val = val-10;
            }
            ListNode item = new ListNode(val);
            node.next = item;
            node = node.next;
            node1 = node1.next;
            node2 = node2.next;
        }
        if (node1 != null){
            node.next = node1;
        }
        if (node2 != null){
            node.next = node2;
        }
        while (add){
            if (node.next == null){
                ListNode listNode = new ListNode(1);
                node.next = listNode;
                break;
            }else {
                node = node.next;
                if (node.val == 9){
                    node.val = 0;
                    add = true;
                }else{
                    node.val = node.val+1;
                    break;
                }
            }
        }
        return pre.next;
    }
}
